#!/usr/bin/env python3
# 项目停止脚本

import subprocess
from pathlib import Path

def run_command(cmd, cwd=None):
    """执行命令"""
    print(f"执行: {cmd}")
    result = subprocess.run(cmd, shell=True, cwd=cwd)
    return result.returncode == 0

def main():
    """主函数"""
    project_root = Path(__file__).parent.parent
    
    print("🛑 停止音视频转换微服务...")
    
    # 停止服务
    if run_command("docker-compose down", cwd=project_root):
        print("✅ 服务已停止")
    else:
        print("❌ 停止服务失败")
    
    # 可选：清理资源
    print("\n🧹 清理选项:")
    print("1. 保留数据 (推荐)")
    print("2. 清理所有数据和镜像")
    
    choice = input("请选择 (1/2): ").strip()
    
    if choice == "2":
        print("🗑️ 清理数据和镜像...")
        run_command("docker-compose down -v --rmi all", cwd=project_root)
        run_command("docker system prune -f", cwd=project_root)
        print("✅ 清理完成")
    else:
        print("✅ 数据已保留")

if __name__ == "__main__":
    main()